little improvement for previous solution

Signed-off-by: Mory Mobile <mory.mobile@gmail.com>