Fixed first unit test