Forgot to add the class to both cases. See (#4af45ab).