Fix pt 2