I should write unit tests