Added edge case for a single div