fix unit test and styling