Fixing the failed test