restoring broken test