Adam: Optimization Fix

Fixed optimization issues