Minor patch

Made an error at previous fix