small fix
Fix pt 2