fixed test so it works