Minor Bug Fix Setup

Fixed minor bug with setup