Fix error during testing