Index: kernel/arch/amd64/src/kseg.c
===================================================================
--- kernel/arch/amd64/src/kseg.c	(revision 8a1be7692f8710c1044d21f081dc5cffc0d5c33d)
+++ kernel/arch/amd64/src/kseg.c	(revision fac0ac7be7d127e6cc5c186e3c497bea39ed2e3b)
@@ -47,5 +47,5 @@
 	kseg_t *kseg;
 
-	kseg = (kseg_t *) malloc(sizeof(kseg_t), FRAME_ATOMIC);
+	kseg = (kseg_t *) malloc(sizeof(kseg_t));
 	if (!kseg)
 		panic("Cannot allocate kseg.");
Index: kernel/arch/amd64/src/pm.c
===================================================================
--- kernel/arch/amd64/src/pm.c	(revision 8a1be7692f8710c1044d21f081dc5cffc0d5c33d)
+++ kernel/arch/amd64/src/pm.c	(revision fac0ac7be7d127e6cc5c186e3c497bea39ed2e3b)
@@ -273,5 +273,5 @@
 		write_cr3((uintptr_t) AS_KERNEL->genarch.page_table);
 
-		tss_p = (tss_t *) malloc(sizeof(tss_t), FRAME_ATOMIC);
+		tss_p = (tss_t *) malloc(sizeof(tss_t));
 		if (!tss_p)
 			panic("Cannot allocate TSS.");