Cleaned up unit test