Fix fatal error on test