Un-borked (minor bug fix)