bug fix
stage 1 complete