Quick bug-fix