- Small fix to simulation