diff --git a/tests/test_flask.py b/tests/test_flask.py
index da9e1c6..758952d 100644
--- a/tests/test_flask.py
+++ b/tests/test_flask.py
@@ -1,6 +1,7 @@
 import unittest
 from flask import Flask, json
 from wsmeext.flask import signature
+from wsme.api import Response
 from wsme.types import Base, text
 
 
@@ -51,6 +52,18 @@ def post_model(body):
     return Model(name=body.name)
 
 
+@test_app.get('/status_sig')
+@signature(int, status_code=201)
+def get_status_sig():
+    return 201
+
+
+@test_app.get('/status_response')
+@signature(int)
+def get_status_response():
+    return Response(1, status_code=201)
+
+
 class FlaskrTestCase(unittest.TestCase):
 
     def setUp(self):