Patching minor mistake