Change tests to use custom user model