missed bit of code in last commit