diff --git a/ColoredSquare.c b/ColoredSquare.c
index 41b9417..da70431 100644
--- a/ColoredSquare.c
+++ b/ColoredSquare.c
@@ -1,5 +1,5 @@
 #include <stdlib.h>
-#include <assert.h>
+#include <stdio.h>
 
 #include "ColoredSquare.h"
 
diff --git a/Figure.h b/Figure.h
index 5c2492c..f67462d 100644
--- a/Figure.h
+++ b/Figure.h
@@ -1,8 +1,6 @@
 #ifndef __FIGURE_H__
 #define __FIGURE_H__
 
-#include <stdio.h>
-
 #include "Object.h"
 
 struct Figure;
diff --git a/Square.c b/Square.c
index 2a38df1..0c34c80 100644
--- a/Square.c
+++ b/Square.c
@@ -1,4 +1,5 @@
 #include <stdlib.h>
+#include <stdio.h>
 
 #include "Square.h"
 
diff --git a/main.c b/main.c
index 4351d91..79b3d72 100644
--- a/main.c
+++ b/main.c
@@ -1,5 +1,4 @@
 #include <stdio.h>
-#include <stdlib.h>
 
 #include "ColoredSquare.h"