Bugfix: Forgot some code