Fix lesson four with detached head solution