fixed a bug from last push