fixed test and assert