small update in logic :beer: