Use assertTrue instead of assertGreater in tests