Whoops. Forgot to remove test code