Fix #2 - left over from last commit