fixed minor bug again :)