Fix for previous commit

Change-Id: I94c59ed7646bd2de6f3f4e4c105335f3563c1253
Signed-off-by: Andrey Shamis <lolnik@gmail.com>