Whoops, did need that little piece of code after all