Just a lil tweak for correctness