bit of bugfix