fix mistake on test