Forgot to lint this one
Should instead of assert
Improved testing