added fix to last commit