minor bug fix. all good now