Fixed failing test due to not mocked attribute