Arguably a nicer solution
Actually this feels less hacky