Improve code for v1.1.0#2