Finally got one unittest to pass