Correcting a bug

Resolve a bug