continuing fixing the last commit

Signed-off-by: RayAllen <ray@ray-Lenovo-IdeaPad-S400.(none)>