Small fix, how did this work before?