Hacky fixes, TODO