Lin fixes Issue 2