Fix for failed test