minor bug sorry :)