Fixing annotation on test