Bug fixed

Added a missing line