Fixing minor issue