Could you please add curly brackets around "return done();" block, and also ensure new behavior is covered with the test case.