Forgot to import assert module