Fixed the error that the willExitState wasn't using
the new done() function