Fixed bug introduced in todays work

Fixed bug introduced in todays work