Fixed error that I didn't catch before