fixed test to run correctly