Fixed unit tests to compare classes correctly