[fix] This should really be at the top