Update for Caustic changes

Signed-off-by:
Aleksi Sapon <QcTechs@gmail.com>