diff --git a/api/v1/serializers/course.py b/api/v1/serializers/course.py
index f04b81ac..d348ce64 100644
--- a/api/v1/serializers/course.py
+++ b/api/v1/serializers/course.py
@@ -175,13 +175,13 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                 if 'id' in material:
                     m = Material.objects.get(id=material['id'])
                     m.title = material['title']
-                    m.cover = ImageObject.objects.get(material['cover'])
+                    m.cover = ImageObject.objects.get(id=material['cover'])
                     m.short_description = material['short_description']
                     m.save()
                 else:
                     m = Material.objects.create(
                         title=material['title'],
-                        cover=ImageObject.objects.get(material['cover']),
+                        cover=ImageObject.objects.get(id=material['cover']),
                         short_description=material['short_description'],
                     )
         return course
@@ -259,13 +259,13 @@ class CourseCreateSerializer(serializers.ModelSerializer):
                 if 'id' in material:
                     m = Material.objects.get(id=material['id'])
                     m.title = material['title']
-                    m.cover = ImageObject.objects.get(material['cover'])
+                    m.cover = ImageObject.objects.get(id=material['cover'])
                     m.short_description = material['short_description']
                     m.save()
                 else:
                     m = Material.objects.create(
                         title=material['title'],
-                        cover=ImageObject.objects.get(material['cover']),
+                        cover=ImageObject.objects.get(id=material['cover']),
                         short_description=material['short_description'],
                     )
                     course.materials.add(m)