TODO: Find a better way to control this