slight fix to last commit