Quick and dirty fix

Just pushing a pull request to myself