minor fix to unit 1