Fix a formatting and add a missing test