slight fixes again :)