Fix for test passing