Bugfix, forgot a few lines