Custom hack. Will implement better later.