add badges and fix a test