First stab at fixing a test