fixed test a bit