Yes this was the one I suspected. 
I'd rather do a if-else but anything works