Fix for test