Bug fix... oops