fixed formatting in tests and added assertion