Forgot to remove test code