make this as balance and change the message in assert equals