Small bug fix
small bug fix