Should add a test case in test_utils.py?