TESTS: Fixed bugs in a test to work with an added ->first()