Fix problem with a test