diff --git a/server/first_core/disassembly/__init__.py b/server/first_core/disassembly/__init__.py
index 87f950b..6515cad 100644
--- a/server/first_core/disassembly/__init__.py
+++ b/server/first_core/disassembly/__init__.py
@@ -72,6 +72,8 @@
 from capstone.mips import MIPS_OP_MEM
 from capstone.mips import MIPS_OP_INVALID
 
+from capstone import Cs
+
 
 arch_mapping = {
     'ppc' : (CS_ARCH_PPC, CS_MODE_32),
diff --git a/server/www/views.py b/server/www/views.py
index 089658d..34ae43f 100644
--- a/server/www/views.py
+++ b/server/www/views.py
@@ -12,7 +12,7 @@
 from first_core.models import Function, User
 from first_core.auth import Authentication, require_login, FIRSTAuthError
 
-def handler404(request):
+def handler404(request, exc):
     return render(request, 'www/404.html', None)