Minor bugfix. Stupid me.