small fix after debugging