diff --git a/lib/src/model/json/message.dart b/lib/src/model/json/message.dart
index bd5f90b..6253d7d 100644
--- a/lib/src/model/json/message.dart
+++ b/lib/src/model/json/message.dart
@@ -88,9 +88,9 @@ class TextContent extends MessageContent {
 
 @JsonSerializable()
 class ImageContent extends MessageContent {
-  final String imageUrl;
+  final String imageToken;
 
-  ImageContent(this.imageUrl);
+  ImageContent(this.imageToken);
 
   factory ImageContent.fromJson(Map<String, dynamic> json) =>
       _$ImageContentFromJson(json);
diff --git a/lib/src/model/json/message.g.dart b/lib/src/model/json/message.g.dart
index ee41642..0d45008 100644
--- a/lib/src/model/json/message.g.dart
+++ b/lib/src/model/json/message.g.dart
@@ -47,10 +47,10 @@ Map<String, dynamic> _$TextContentToJson(TextContent instance) =>
     };
 
 ImageContent _$ImageContentFromJson(Map<String, dynamic> json) => ImageContent(
-      json['imageUrl'] as String,
+      json['imageToken'] as String,
     );
 
 Map<String, dynamic> _$ImageContentToJson(ImageContent instance) =>
     <String, dynamic>{
-      'imageUrl': instance.imageUrl,
+      'imageToken': instance.imageToken,
     };
diff --git a/lib/src/views/register_view.dart b/lib/src/views/register_view.dart
index cfe0d28..2dac9b3 100644
--- a/lib/src/views/register_view.dart
+++ b/lib/src/views/register_view.dart
@@ -1,5 +1,4 @@
 import 'package:twonly/main.dart';
-import 'package:twonly/src/providers/api_provider.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter/services.dart';
 import 'package:flutter_gen/gen_l10n/app_localizations.dart';