Fixed error with test