typo fix
fixed the tests