Eh I give up... TODO fix this...