yeah, could prob do with a unit test