making small fix to beta proposed code