Just a really quick hack to make this work.