It works now, but a test fails