diff --git a/selenium/end-to-end.spec.js b/selenium/end-to-end.spec.js
index c09243b..68f25de 100644
--- a/selenium/end-to-end.spec.js
+++ b/selenium/end-to-end.spec.js
@@ -282,7 +282,7 @@ describe("Selenium End to End tests.", function () {
       // 22 | click | css=.css-16nf58z |
       await studentDriver.findElement(By.css(".css-16nf58z")).click();
       // 23 | Check that the student attended successfully
-      // await studentDriver.sleep(1000); // Need to wait for the modal to show up
+       await studentDriver.sleep(1000); // Need to wait for the modal to show up
       const failureMessage = await studentDriver
         .findElement(By.id("toast-1-title"))
         .getText();
diff --git a/src/components/primary-button/PrimaryButton.tsx b/src/components/primary-button/PrimaryButton.tsx
index b71f057..0e2e413 100644
--- a/src/components/primary-button/PrimaryButton.tsx
+++ b/src/components/primary-button/PrimaryButton.tsx
@@ -1,4 +1,4 @@
-import { Button } from "@chakra-ui/react";
+import { Button, Container } from "@chakra-ui/react";
 
 interface PrimaryButtonProps {
   children: string | JSX.Element;
@@ -14,8 +14,8 @@ export function PrimaryButton({
   onClick = () => {},
 }: PrimaryButtonProps) {
   return (
-    <Button variant={"primary"} width={width} size={size} onClick={onClick}>
+    <Container variant={"primary"} width={width} size={size} onClick={onClick}>
       {children}
-    </Button>
+    </Container>
   );
 }