Fixed a niggle