Fixed a bug. had a test on two places.