Fixing the failing test