Forgot to uncomment test logic