Fixing small error