Small refactoring action

Signed-off-by: Andrew Tselischev <andrewtselischev@gmail.com>