fixing  broken test