diff --git a/tests/test_todo_parser.py b/tests/test_todo_parser.py
index df25c9c..a4c91d7 100644
--- a/tests/test_todo_parser.py
+++ b/tests/test_todo_parser.py
@@ -2,7 +2,7 @@ import json
 import os
 import unittest
 
-from main import TodoParser
+from TodoParser import TodoParser
 
 
 def count_issues_for_file_type(raw_issues, file_type):