Can you add a test to functional_tests.py that this works?