Minor Bug fix
My fault!