Pushing fix for failing test