Fixes small error in last commit