Fix first simple unit test