Mock does not return different types