fixed bug passing tests now