Fixed my tests breaking!