Fix for error in last commit