Fixing class namespaces in unit test