1 files changed, 1 insertions, 1 deletions
diff --git a/Classes/ChatViewController.m b/Classes/ChatViewController.m
index 971363c..5a9f01b 100644
--- a/Classes/ChatViewController.m
+++ b/Classes/ChatViewController.m
@@ -40,7 +40,7 @@
     self.view.backgroundColor = [UIColor colorWithRed:219.0f/255.0f green:226.0f/255.0f blue:237.0f/255.0f alpha:1];
 	
 	chatTableView = [[UITableView alloc] initWithFrame:self.view.bounds style:UITableViewStylePlain];
-	dataSource = [[ChatMessagesDataSource alloc] init];
+	dataSource = [[ChatMessagesDataSource alloc] initWithTableView:chatTableView];
 	[dataSource startMyWebSocket];
 	[chatTableView setDataSource:dataSource];
 	[self.view addSubview:chatTableView];