fix mistake on test class