Fixing an edge Case