Should have run tests... fixed bug