Fixing this, was too trivial, don't know how I missed.