fix compile issue - attempt 2