Fixing the test to check equality and not assign